package com.guocheng.modules.login.utils;

import com.guocheng.modules.login.model.dto.DeptDTO;
import com.guocheng.modules.login.model.po.Dept;

import java.util.List;
import java.util.stream.Collectors;

public class DeptUtils {

    /**
     * 部门实体类转换工具类
     * @param depts
     * @return
     */
    public static List<DeptDTO> deptToDTO(List<Dept> depts){
        return depts.stream().map(dept -> deptDTO(dept)).collect(Collectors.toList());
    }

    /**
     * 部门实体类转换工具类
     * @param dept
     * @return
     */
    public static DeptDTO deptDTO(Dept dept){
        return new DeptDTO(dept.getId(),dept.getParentId(), dept.getName(),dept.getDescription());
    }
}
